
if __name__ == '__main__':
    learning_rates = [1e-3, 1e-4, 5e-5, 1e-5]
